package br.edu.ufcg.greengrid.model.machine.state.factory;

import java.util.Map;

import br.edu.ufcg.greengrid.model.machine.state.GreenifiableIdleMachineState;
import br.edu.ufcg.greengrid.model.machine.state.MachineState;
import br.edu.ufcg.greengrid.model.machine.state.StandbyGoingGreenMachineState;
import br.edu.ufcg.greengrid.model.machine.state.StandbyLeavingGreenMachineState;
import br.edu.ufcg.greengrid.model.machine.state.MachineState.MSEnum;

public class StandByMachineStateFactory implements MachineStateFactory {

	@Override
	public Map<MSEnum, MachineState> getStates() {

		Map<MSEnum, MachineState> states = new NormalMachineStateFactory().getStates();
		states.put(MSEnum.IDLE, new GreenifiableIdleMachineState());
		states.put(MSEnum.GOING_GREEN, new StandbyGoingGreenMachineState());
		states.put(MSEnum.LEAVING_GREEN, new StandbyLeavingGreenMachineState());
		return states;
	}

}
